package com.xuecheng.auth.config;


import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.oauth2.provider.token.AuthorizationServerTokenServices;
import org.springframework.security.oauth2.provider.token.DefaultTokenServices;
import org.springframework.security.oauth2.provider.token.TokenEnhancerChain;
import org.springframework.security.oauth2.provider.token.TokenStore;
import org.springframework.security.oauth2.provider.token.store.JwtAccessTokenConverter;
import org.springframework.security.oauth2.provider.token.store.JwtTokenStore;

import javax.annotation.Resource;
import java.util.Arrays;

/**
 * @author Administrator
 * @version 1.0
 **/
@Configuration

public class TokenConfig {

    private final String SIGNING_KEY = "mq123";

    @Resource
    private TokenStore tokenStore; //下面定义Bean
    @Resource
    private JwtAccessTokenConverter accessTokenConverter; //下面定义Bean

    /*@Bean
    public TokenStore tokenStore() {
    //使用内存存储令牌（普通令牌）
    return new InMemoryTokenStore();
        }
    */
    //设置签名密钥
    @Bean
    public JwtAccessTokenConverter accessTokenConverter() {
        JwtAccessTokenConverter converter = new JwtAccessTokenConverter();
        converter.setSigningKey(SIGNING_KEY);
        return converter;
    }

    //使用使用jwt令牌方式存储令牌
    @Bean
    public TokenStore tokenStore() {
        //使用使用jwt令牌方式存储令牌
        return new JwtTokenStore(accessTokenConverter);
    }


    //令牌管理服务
    @Bean(name = "authorizationServerTokenServicesCustom")
    public AuthorizationServerTokenServices tokenService() {
        DefaultTokenServices service = new DefaultTokenServices();
        service.setSupportRefreshToken(true);//支持刷新令牌
        service.setTokenStore(tokenStore);//令牌存储策略

        //控制 JWT token 增强器链（Token Enhancer）的类。使用 Token Enhancer
        // 可以给 JWT token 添加自定义信息，如用户权限、用户名等等，增强 token 的功能
        TokenEnhancerChain tokenEnhancerChain = new TokenEnhancerChain();
        tokenEnhancerChain.setTokenEnhancers(Arrays.asList(accessTokenConverter));//将 accessTokenConverter 包装成一个列表，并设置到 TokenEnhancerChain
        service.setTokenEnhancer(tokenEnhancerChain);

        service.setAccessTokenValiditySeconds(7200); // 令牌默认有效期2小时
        service.setRefreshTokenValiditySeconds(259200); // 刷新令牌默认有效期3天
        return service;
    }


}
